import requests


class TiebaSpider:
    def __init__(self, tieba_name):
        self.tieba_name = tieba_name
        self.url_temp = "https://tieba.baidu.com/f?kw=%s&ie=utf-8&pn={}" % tieba_name
        self.headers = {
            "User-Agent":"Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.77 Mobile Safari/537.36"
        }

    def get_url_list(self):
        """构造url列表"""
        # url_list = []
        # for i in range(4):
        #     url_list.append(self.url_temp.format(i * 50))
        #
        # return url_list
        return [self.url_temp.format(i*50) for i in range(4)]

    def parse_url(self, url):
        print(url)
        """发送请求 获取响应"""
        response = requests.get(url, headers=self.headers)
        return response.content.decode()

    def save_html(self, html_str,page_num):
        """保存html字符串"""
        file_path = "{}-第{}页.html".format(self.tieba_name,page_num)
        with open(file_path, "w", encoding="utf-8") as f:
            f.write(html_str)

    def run(self):
        """实现主要逻辑"""
        # 1 构造url列表
        url_list = self.get_url_list()
        # 2 遍历，发送请求，获取响应
        for url in url_list:
            html_str = self.parse_url(url)
            # 3 保存
            page_num = url_list.index(url)+1
            self.save_html(html_str,page_num)


if __name__ == '__main__':
    tieba_spider = TiebaSpider("李毅")
    tieba_spider.run()
